package com.kabuda.zhugehot200;

import java.util.Arrays;

/**
 * @author kabuda
 * @date 2025-03-15 16:08
 * @description
 */
public class tLCR012_寻找数组的中心下标 {
    public static void main(String[] args) {
        System.out.println(pivotIndex2(new int[]{1, 7, 3, 6, 5, 6}));
    }

    public static int pivotIndex(int[] nums) {
        //数组值的总和（右边的总和值）
        int sum = Arrays.stream(nums).sum();
        //当前遍历到的总和（左边的总和值）
        int total = 0;
        for (int i = 0; i < nums.length; i++) {
            //左边的值+当前值
            total += nums[i];
            //如果当前值等于总和，则返回当前下标
            if (total == sum) {
                return i;
            } else {
                sum -= nums[i];
            }
        }
        return -1;

    }

    public static int pivotIndex2(int[] nums) {
        //数组值的总和
        int sum = Arrays.stream(nums).sum();
        //当前遍历到的总和
        int total = 0;
        for (int i = 0; i < nums.length; i++) {
            //如果当前累加值*2 + 当前值 = 总和，则返回当前下标
            if (2 * total + nums[i] == sum) {
                return i;
            }
            total += nums[i];

        }
        return -1;
    }

}
